package com.itheima.stock.helper;

import com.google.common.util.concurrent.RateLimiter;
import com.itheima.stock.enums.CodeEnum;
import com.itheima.stock.response.ResponseResult;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @Author:gzc
 * @Date: 2025/10/23 21:26
 * @Description:
 */
@Aspect
@Component
@SuppressWarnings("all")
public class LimiterAspect {

    @Autowired
    private  RateLimiter rateLimiter;
    @Pointcut("@annotation(com.itheima.stock.annotation.Limiter)")
    public void pt(){}



    @Around("pt()")
    public Object limiter(ProceedingJoinPoint joinPoint) throws Throwable {
        //流量控制
        boolean flag = rateLimiter.tryAcquire();
        if (flag){
            joinPoint.proceed();
        }else {
            return ResponseResult.error(CodeEnum.OVER_CONCURRENT_ERROR);
        }

        return joinPoint.proceed();
    }
}
